from tensorflow.keras import *


def read_imdb(max_features):
    ds = datasets.imdb.load_data(num_words=max_features)
    return ds

# dataset = read_imdb(20000)